# Adapted from Python 2.7 sqlalchemy-1.013/sql/crud.py
# Bug was in handling generator comprehension
# In 2.6, 2.7 JUMP_ABSOLUTEs rather than JUMP_LOOPs are generated

# RUNNABLE!
def extend(stmt, a, c, c1, c2, c3):
    return c(([(5 if c1 else c2) if a else c3] for i in enumerate(stmt)))


def foo(gen):
    return list(gen)


assert extend([0], 0, foo, True, "c2", "c3") == [["c3"]]
assert extend([0, 1], 1, foo, False, "c2", "c3") == [["c2"], ["c2"]]
assert extend([0, 1], False, foo, False, "c2", "c3") == [["c3"], ["c3"]]
